"""
给定一个二叉树，找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。

示例：
给定二叉树 [3,9,20,null,null,15,7]，

    3
   / \
  9  20
    /  \
   15   7
返回它的最大深度 3 。

作者：力扣 (LeetCode)
链接：https://leetcode-cn.com/leetbook/read/data-structure-binary-tree/xoh1zg/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
"""


class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution:
    # 自底向上
    def maxDepth(self, root: TreeNode) -> int:
        if not root:
            return 0
        left_depth = self.maxDepth(root.left)
        right_depth = self.maxDepth(root.right)
        return max(left_depth, right_depth) + 1

    # 自顶向下
    def max_depth(self, root: TreeNode) -> int:
        if not root:
            return self.answer
        self.depth(root, 1)
        return self.answer

    def depth(self, node: TreeNode, depth: int):
        if not node:
            return
        if node.left is None and node.right is None:
            self.answer = max(self.answer, depth)
        if node.left:
            self.depth(node.left, depth+1)
        if node.right:
            self.depth(node.right, depth+1)

    answer = 0


if __name__ == '__main__':
    tree = TreeNode(3)
    tree.left = TreeNode(9)
    tree.right = TreeNode(20)
    tree.right.left = TreeNode(15)
    tree.right.right = TreeNode(7)
    s = Solution()
    print(s.maxDepth(tree))
    print(s.max_depth(tree))
